import torch

a = torch.tensor([1, 2, 3, 4, 5, 6]).float()
c = torch.tensor([[1, 2, 3, 4, 5, 6]]).float()

b = torch.repeat_interleave(a, 3, dim=0)
print(b)
e = torch.repeat_interleave(c, 3, dim=0)
print(e)